import torch
torch.set_num_threads(1)

# 设备配置
DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
LABEL_COL = "cogimp_label"
# 其他配置可以在这里添加
BATCH_SIZE = 32
LEARNING_RATE = 0.001
EPOCHS = 10
DROPOUT = 0.5
EMBEDDING_DIM = 8
HIDDEN_DIM = 128
# transformer特有的
HEADS = 4
DEPTH = 3
ATTN_DROPOUT = 0.5
# mlp特有的
HIDDEN_DIMS_MLP = [256, 128]
# lstm特有的
NUM_LAYERS = 1
BIDIRECTIONAL = True

